def demo(a):
    if a == 1:
        return "a"
    elif a == 2:
        return "b"
